package org.strangewt.demo.composit;

import org.strangewt.demo.testdata.Person;
import org.strangewt.demo.testdata.Project;
import org.strangewt.jface.viewers.ICompositeProvider;

public class CompositeProviderPersonProject implements ICompositeProvider {

	@Override
	public Class<?> getCompositeClass(Object o) {
		
		if(o.getClass() == Person.class) {
			return CompositePerson.class;
		}
		if(o.getClass() == Project.class) {
			return CompositeProject.class;
		}
		
		return null;
	}
}
